package luo.hui.jiang.recursion;

import java.util.Queue;

/**
 * 八皇后问题 回溯法
 *
 * @author 罗惠江
 * @version 1.0
 * @email 1511619294@qq.com
 * @date 2020/7/29 21:12
 */
public class Queue8 {

    // 定义一个max表示公有多少个换皇后
    int max = 8;
    // 定义数组 array ； 保存皇后放置位置的结果，比如 arr = {0,4,7,5,2,6,1,3}
    int[] array = new int[max];
    static int count = 0;
    static int judgeCount = 0;

    public static void main(String[] args) {
        // 测试一把， 8 皇后是否正确
        Queue8 queue8 = new Queue8();
        queue8.check(0);
        System.out.println(count);
        System.out.println(judgeCount);

    }

    private void check(int n) {
        if (n == max) { // n = 8 ,其中 8 个皇后就放好了
            print();
            return;
        }
        // 依次放入皇后，并判断是否冲突
        for (int i = 0; i < max; i++) {
            // 先把当前这个皇后n，放到该行的第一列
            array[n] = i;
            // 判断当前放置第n 个皇后到i列时，是否冲突
            if (judge(n)) {
                // b不冲突
                // 接着放n + 1个 皇后，即开始递归。
                check(n + 1);
            }
        }
    }

    private boolean judge(int n) {
        judgeCount++;
        for (int i = 0; i < n; i++) {
            // 1.array[i] == array[n] 表示判断 第 n 个 换后是否和前面的n-1个换后在同一列
            // 2.Math.abs(n-i) == Math.abs(array[n]-array[i]) 判断是不是在同一个斜线上
            // 3.没有必要判断是否在同一行，我们不断递增的设计，已经避免了这种情况的发生
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
                return false;
            }
        }
        return true;
    }

    private void print() {
        count++;
        for (int i : array) {
            System.out.print(i + " ");
        }
        System.out.println();
    }

}
